using SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace SqlInjectionScanner.Models
{
    // InjectionTestData.cs
    [SugarTable("injection_test_data")]
    public class InjectionTestData
    {
        [SugarColumn(IsPrimaryKey = true, IsIdentity = true)]
        public int Id { get; set; }
        
        [SugarColumn(IsNullable = false)]
        public string Content { get; set; } = string.Empty;
        
        [SugarColumn(IsNullable = false)]
        public string Type { get; set; } = string.Empty;
    }

    // InjectionRule.cs
    [SugarTable("injection_rules")]
    public class InjectionRule
    {
        [SugarColumn(IsPrimaryKey = true, IsIdentity = true)]
        public int Id { get; set; }
        
        [SugarColumn(IsNullable = false)]
        public string Name { get; set; } = string.Empty;
        
        [SugarColumn(IsNullable = false)]
        public string RegexPattern { get; set; } = string.Empty;
        
        [SugarColumn(IsNullable = false)]
        public string RiskLevel { get; set; } = string.Empty;
        
        [SugarColumn(IsNullable = false)]
        public string InjectionType { get; set; } = string.Empty;
        
        [SugarColumn(IsNullable = false)]
        public bool IsEnabled { get; set; } = true;
    }
}
